Beyond Repeated Sampling: Learning Search Policies for LLM Reasoning
超越重复采样:学习用于LLM推理的搜索策略
🧭 LLM推理 · 可训练搜索策略 ★ 深度解读
Meta FAIR · Université Paris-Saclay · Inria · CNRS · NYU Courant Institute and CDS
用RL训练小型概念生成器作为大模型推理的搜索策略,在hard数学问题上pass@128翻倍并跨模型迁移
📌 工作参考价值(4 点)
- 推理时多样性不要只调 temperature/top-p,可以在 prompt 层注入多个高层概念、策略或意图,并用单轨迹批量生成,避免近重复采样;对应推荐/广告中可对同一 query 生成多个语义方向再分别生成候选文案或推荐理由。
- 训练小型 orchestrator/概念生成器作为黑盒或闭源大模型的搜索策略,RL 奖励直接来自下游大模型的 pass@k;max-of-mean 比 max-of-max 信息量更大、更稳定,适合作为奖励聚合方式。
- 训练成本主要在下游大模型 rollout 和 judge,可以用固定 judge 复用并离线采样;在业务中训练轻量 agent 来引导重模型探索,不必微调大模型,且学习到的策略可跨模型家族迁移。
- 实验设计要公平:基线必须使用高温度、高 top-p 的探索性解码,否则概念引导或策略提升的效果可能虚高;报告 pass@k 时要匹配相同的 answer generation 预算。